home *** CD-ROM | disk | FTP | other *** search
/ Clickx 47 / Clickx 47.iso / assets / software / Miro_Installer.exe / xulrunner / python / psyco / core.py < prev    next >
Encoding:
Python Source  |  2007-04-13  |  8.2 KB  |  233 lines
  1. ###########################################################################
  2. # 
  3. #  Psyco main functions.
  4. #   Copyright (C) 2001-2002  Armin Rigo et.al.
  5.  
  6. """Psyco main functions.
  7.  
  8. Here are the routines that you can use from your applications.
  9. These are mostly interfaces to the C core, but they depend on
  10. the Python version.
  11.  
  12. You can use these functions from the 'psyco' module instead of
  13. 'psyco.core', e.g.
  14.  
  15.     import psyco
  16.     psyco.log('/tmp/psyco.log')
  17.     psyco.profile()
  18. """
  19. ###########################################################################
  20.  
  21. import _psyco
  22. import types, new
  23. from support import *
  24.  
  25.  
  26. # Default charge profiler values
  27. default_watermark     = 0.09     # between 0.0 (0%) and 1.0 (100%)
  28. default_halflife      = 0.5      # seconds
  29. default_pollfreq_profile    = 20       # Hz
  30. default_pollfreq_background = 100      # Hz -- a maximum for sleep's resolution
  31. default_parentframe   = 0.25     # should not be more than 0.5 (50%)
  32.  
  33.  
  34. def full(memory=None, time=None, memorymax=None, timemax=None):
  35.     """Compile as much as possible.
  36.  
  37. Typical use is for small scripts performing intensive computations
  38. or string handling."""
  39.     import profiler
  40.     if PYTHON_SUPPORT:
  41.         p = profiler.FullCompiler()
  42.     else:
  43.         p = profiler.ActiveProfiler(0.0, 0.5)
  44.     p.run(memory, time, memorymax, timemax)
  45.  
  46.  
  47. def profile(watermark   = default_watermark,
  48.             halflife    = default_halflife,
  49.             pollfreq    = default_pollfreq_profile,
  50.             parentframe = default_parentframe,
  51.             memory=None, time=None, memorymax=None, timemax=None):
  52.     """Turn on profiling.
  53.  
  54. The 'watermark' parameter controls how easily running functions will
  55. be compiled. The smaller the value, the more functions are compiled."""
  56.     import profiler
  57.     p = profiler.ActivePassiveProfiler(watermark, halflife,
  58.                                        pollfreq, parentframe)
  59.     p.run(memory, time, memorymax, timemax)
  60.  
  61.  
  62. def background(watermark   = default_watermark,
  63.                halflife    = default_halflife,
  64.                pollfreq    = default_pollfreq_background,
  65.                parentframe = default_parentframe,
  66.                memory=None, time=None, memorymax=None, timemax=None):
  67.     """Turn on passive profiling.
  68.  
  69. This is a very lightweight mode in which only intensively computing
  70. functions can be detected. The smaller the 'watermark', the more functions
  71. are compiled."""
  72.     import profiler
  73.     p = profiler.PassiveProfiler(watermark, halflife, pollfreq, parentframe)
  74.     p.run(memory, time, memorymax, timemax)
  75.  
  76.  
  77. def runonly(memory=None, time=None, memorymax=None, timemax=None):
  78.     """Nonprofiler.
  79.  
  80. XXX check if this is useful and document."""
  81.     if PYTHON_SUPPORT:
  82.         import profiler
  83.         p = profiler.RunOnly()
  84.         p.run(memory, time, memorymax, timemax)
  85.  
  86.  
  87. def stop():
  88.     """Turn off all automatic compilation.  bind() calls remain in effect."""
  89.     import profiler
  90.     profiler.go([])
  91.  
  92.  
  93. def log(logfile='', mode='w', top=10):
  94.     """Enable logging to the given file.
  95.  
  96. If the file name is unspecified, a default name is built by appending
  97. a 'log-psyco' extension to the main script name.
  98.  
  99. Mode is 'a' to append to a possibly existing file or 'w' to overwrite
  100. an existing file. Note that the log file may grow quickly in 'a' mode."""
  101.     import profiler, logger
  102.     if not logfile:
  103.         import os
  104.         logfile, dummy = os.path.splitext(sys.argv[0])
  105.         if os.path.basename(logfile):
  106.             logfile += '.'
  107.         logfile += 'log-psyco'
  108.     if hasattr(_psyco, 'VERBOSE_LEVEL'):
  109.         print >> sys.stderr, 'psyco: logging to', logfile
  110.     # logger.current should be a real file object; subtle problems
  111.     # will show up if its write() and flush() methods are written
  112.     # in Python, as Psyco will invoke them while compiling.
  113.     logger.current = open(logfile, mode)
  114.     logger.print_charges = top
  115.     profiler.logger = logger
  116.     logger.writedate('Logging started')
  117.     cannotcompile(logger.psycowrite)
  118.     _psyco.statwrite(logger=logger.psycowrite)
  119.  
  120.  
  121. def bind(x, rec=None):
  122.     """Enable compilation of the given function, method, or class object.
  123.  
  124. If C is a class (or anything with a '__dict__' attribute), bind(C) will
  125. rebind all functions and methods found in C.__dict__ (which means, for
  126. classes, all methods defined in the class but not in its parents).
  127.  
  128. The optional second argument specifies the number of recursive
  129. compilation levels: all functions called by func are compiled
  130. up to the given depth of indirection."""
  131.     if isinstance(x, types.MethodType):
  132.         x = x.im_func
  133.     if isinstance(x, types.FunctionType):
  134.         if rec is None:
  135.             x.func_code = _psyco.proxycode(x)
  136.         else:
  137.             x.func_code = _psyco.proxycode(x, rec)
  138.         return
  139.     if hasattr(x, '__dict__'):
  140.         funcs = [o for o in x.__dict__.values()
  141.                  if isinstance(o, types.MethodType)
  142.                  or isinstance(o, types.FunctionType)]
  143.         if not funcs:
  144.             raise error, ("nothing bindable found in %s object" %
  145.                           type(x).__name__)
  146.         for o in funcs:
  147.             bind(o, rec)
  148.         return
  149.     raise TypeError, "cannot bind %s objects" % type(x).__name__
  150.  
  151.  
  152. def unbind(x):
  153.     """Reverse of bind()."""
  154.     if isinstance(x, types.MethodType):
  155.         x = x.im_func
  156.     if isinstance(x, types.FunctionType):
  157.         try:
  158.             f = _psyco.unproxycode(x.func_code)
  159.         except error:
  160.             pass
  161.         else:
  162.             x.func_code = f.func_code
  163.         return
  164.     if hasattr(x, '__dict__'):
  165.         for o in x.__dict__.values():
  166.             if (isinstance(o, types.MethodType)
  167.              or isinstance(o, types.FunctionType)):
  168.                 unbind(o)
  169.         return
  170.     raise TypeError, "cannot unbind %s objects" % type(x).__name__
  171.  
  172.  
  173. def proxy(x, rec=None):
  174.     """Return a Psyco-enabled copy of the function.
  175.  
  176. The original function is still available for non-compiled calls.
  177. The optional second argument specifies the number of recursive
  178. compilation levels: all functions called by func are compiled
  179. up to the given depth of indirection."""
  180.     if isinstance(x, types.FunctionType):
  181.         if rec is None:
  182.             code = _psyco.proxycode(x)
  183.         else:
  184.             code = _psyco.proxycode(x, rec)
  185.         return new.function(code, x.func_globals, x.func_name)
  186.     if isinstance(x, types.MethodType):
  187.         p = proxy(x.im_func, rec)
  188.         return new.instancemethod(p, x.im_self, x.im_class)
  189.     raise TypeError, "cannot proxy %s objects" % type(x).__name__
  190.  
  191.  
  192. def unproxy(proxy):
  193.     """Return a new copy of the original function of method behind a proxy.
  194. The result behaves like the original function in that calling it
  195. does not trigger compilation nor execution of any compiled code."""
  196.     if isinstance(proxy, types.FunctionType):
  197.         return _psyco.unproxycode(proxy.func_code)
  198.     if isinstance(proxy, types.MethodType):
  199.         f = unproxy(proxy.im_func)
  200.         return new.instancemethod(f, proxy.im_self, proxy.im_class)
  201.     raise TypeError, "%s objects cannot be proxies" % type(proxy).__name__
  202.  
  203.  
  204. def cannotcompile(x):
  205.     """Instruct Psyco never to compile the given function, method
  206. or code object."""
  207.     if isinstance(x, types.MethodType):
  208.         x = x.im_func
  209.     if isinstance(x, types.FunctionType):
  210.         x = x.func_code
  211.     if isinstance(x, types.CodeType):
  212.         _psyco.cannotcompile(x)
  213.     else:
  214.         raise TypeError, "unexpected %s object" % type(x).__name__
  215.  
  216.  
  217. def dumpcodebuf():
  218.     """Write in file psyco.dump a copy of the emitted machine code,
  219. provided Psyco was compiled with a non-zero CODE_DUMP.
  220. See py-utils/httpxam.py to examine psyco.dump."""
  221.     if hasattr(_psyco, 'dumpcodebuf'):
  222.         _psyco.dumpcodebuf()
  223.  
  224.  
  225. ###########################################################################
  226. # Psyco variables
  227. #   error         * the error raised by Psyco
  228. #   warning       * the warning raised by Psyco
  229. #   __in_psyco__  * a new built-in variable which is always zero, but which
  230. #                     Psyco special-cases by returning 1 instead. So
  231. #                     __in_psyco__ can be used in a function to know if
  232. #                     that function is being executed by Psyco or not.
  233.